1879D - Sum of XOR Functions - CodeForces Solution


bitmasks combinatorics dp

Please click on ads to support us..

C++ Code:

#include <bits/stdc++.h>
using namespace std;
    
#ifndef ONLINE_JUDGE
#include "debug.hpp"
#else
#define debug(...) 8
#endif
    
typedef long long        ll;
#define f(i, x, n)       for (ll i = x; i < n; i++)
#define rf(i, x, n)      for (ll i = x; i >= n; i--)
#define fr(x,ds)         for(auto &x : ds)
#define pb               push_back
#define all(x)           (x).begin(), (x).end()
    
const ll mod = 998244353;
const ll inf = 1e18;
const ll N = 1e5+5;
    
void solve(){
    ll n;
    cin>>n;
    vector<ll> a(n);
    f(i,0,n) cin>>a[i];
    ll ans=0;
    f(i,0,30) {
        bool ev=0;
        ll even=-1,odd=0;
        ll evencnt=1,oddcnt=0;
        ll sum=0;
        f(j,0,n) {
            if((1ll<<i)&a[j]) {
                ev^=1;
            }
            if(!ev) {
                even+=j;
                evencnt++;
            }
            else {
                odd+=j;
                oddcnt++;
            }
            if(ev) {
                sum+=j*evencnt-even;
            }
            else {
                sum+=j*oddcnt-odd;
            }
        }
        ll tmp=(sum%mod)*((1ll<<i)%mod)%mod;
        ans=(ans+tmp)%mod;
    }
    cout<<ans<<'\n';
}
    
signed main() {
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    ll t = 1;
    // cin >> t;
    for (ll T = 1; T <= t; T++) {
        // cout << "Case #" << T << ": ";
        solve();
    }
    return 0;
}


Comments

Submit
0 Comments
More Questions

1302. Deepest Leaves Sum
1209. Remove All Adjacent Duplicates in String II
994. Rotting Oranges
983. Minimum Cost For Tickets
973. K Closest Points to Origin
969. Pancake Sorting
967. Numbers With Same Consecutive Differences
957. Prison Cells After N Days
946. Validate Stack Sequences
921. Minimum Add to Make Parentheses Valid
881. Boats to Save People
497. Random Point in Non-overlapping Rectangles
528. Random Pick with Weight
470. Implement Rand10() Using Rand7()
866. Prime Palindrome
1516A - Tit for Tat
622. Design Circular Queue
814. Binary Tree Pruning
791. Custom Sort String
787. Cheapest Flights Within K Stops
779. K-th Symbol in Grammar
701. Insert into a Binary Search Tree
429. N-ary Tree Level Order Traversal
739. Daily Temperatures
647. Palindromic Substrings
583. Delete Operation for Two Strings
518. Coin Change 2
516. Longest Palindromic Subsequence
468. Validate IP Address
450. Delete Node in a BST